package android.ijiajiao.util;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.UUID;

import javax.servlet.http.Part;

public class SaveImgUtil {

	public String saveImg(String path, Part part, String savePath) throws IOException {
		File file = null;
		// 判断是否存入成功，如果成功，把图片路径存到数据库中
		String imgPath = null;
		// 文件后缀名
		// 获取图片后缀名
		String contentType = part.getContentType();
		System.out.println(contentType);
		// String fileExt = "."+contentType.substring(6);
		String fileExt = ".jpg";
		// 生成文件名
		String filename = UUID.randomUUID().toString();
		// 把文件写到指定路径
		// 先判断之前是否存在图片，如果有则删除
		file = new File(path + File.separator + filename + fileExt);
		if (file.isFile() && file.exists()) {
			file.delete();
		}
		part.write(path + File.separator + filename + fileExt);
		if (file.isFile() && file.exists()) {
			imgPath = savePath + filename + fileExt;
		}
		return imgPath;
	}

	public ArrayList<String> saveImgs(String path, Collection<Part> parts, String savePath) throws IOException {
		ArrayList<String> img_list = new ArrayList<String>();
		if (!parts.isEmpty()) {
			File file = null;
			String img_path = null;

			for (Part part : parts) {

				if ("file".equals(part.getName())) {
					if (part.getSize() > 0) {
						// 文件后缀名
						// 获取图片后缀名

						String contentType = part.getContentType();
						String fileExt = "." + contentType.substring(6);
						// 生成文件名

						String filename = UUID.randomUUID().toString();
						// 先判断之前是否存在图片，如果有则删除
						file = new File(path + File.separator + filename + fileExt);
						if (file.isFile() && file.exists()) {
							file.delete();
						}
						// 把文件写到指定路径
						part.write(path + File.separator + filename + fileExt);
						if (file.isFile() && file.exists()) {
							img_path = savePath + filename + fileExt;
							img_list.add(img_path);
						}

					}

				}

			}

		}

		return img_list;
	}

}
